package com.suqiu.qqrobot.config;

import com.suqiu.qqrobot.config.properties.BotProperties;
import com.suqiu.qqrobot.service.AiChatService;
import love.forte.simbot.message.ResourceImage;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class BotConfig {

    @Bean
    public BotProperties botProperties() {
        return new BotProperties();
    }

    @Bean
    public AiSelector botSelector(final BotProperties botProperties,
                                  final AiChatService<String, String> openAiDavinciModelService,
                                  final AiChatService<String, String> openAiTurboModelService,
                                  final AiChatService<String, ResourceImage> openAiImageModelService,
                                  final AiChatService<String, String> qingYuanModelService) {
        return new AiSelector(botProperties, openAiDavinciModelService, openAiTurboModelService, qingYuanModelService, openAiImageModelService);
    }

}
